//车上最初有 capacity 个空座位。车 只能 向一个方向行驶（也就是说，不允许掉头或改变方向） 
//
// 给定整数 capacity 和一个数组 trips , trip[i] = [numPassengersi, fromi, toi] 表示第 i 次旅行有
// numPassengersi 乘客，接他们和放他们的位置分别是 fromi 和 toi 。这些位置是从汽车的初始位置向东的公里数。 
//
// 当且仅当你可以在所有给定的行程中接送所有乘客时，返回 true，否则请返回 false。 
//
// 
//
// 示例 1： 
//
// 
//输入：trips = [[2,1,5],[3,3,7]], capacity = 4
//输出：false
// 
//
// 示例 2： 
//
// 
//输入：trips = [[2,1,5],[3,3,7]], capacity = 5
//输出：true
// 
//
// 
//
// 提示： 
//
// 
// 1 <= trips.length <= 1000 
// trips[i].length == 3 
// 1 <= numPassengersi <= 100 
// 0 <= fromi < toi <= 1000 
// 1 <= capacity <= 10⁵ 
// 
// Related Topics 数组 前缀和 排序 模拟 堆（优先队列） 👍 192 👎 0

package leetcode.editor.cn;

class CarPooling {
    public static void main(String[] args) {
        Solution solution = new CarPooling().new Solution();
    }

    //leetcode submit region begin(Prohibit modification and deletion)
    class Solution {
        int[] diff;

        public boolean carPooling(int[][] trips, int capacity) {
            // 由于一开始没有人坐，所以初始化为0
            diff = new int[1001];

            for (int[] trip : trips) {
                // 注意这里在trip[2]乘客已经下车，即乘客在车上的区间是 [trip[1], trip[2] - 1]
                increase(trip[1], trip[2] - 1, trip[0]);
            }

            // 超载直接不满足
            int[] result = result();
            for (int i = 0; i < result.length; i++) {
                if (result[i] > capacity) return false;
            }

            return true;
        }

        public void increase(int i, int j, int val) {
            diff[i] += val;
            if (j + 1 < diff.length) {
                diff[j + 1] -= val;
            }
        }

        public int[] result() {
            int[] res = new int[diff.length];
            res[0] = diff[0];
            for (int i = 1; i < diff.length; i++) {
                res[i] = res[i - 1] + diff[i];
            }

            return res;
        }
    }
//leetcode submit region end(Prohibit modification and deletion)

}
